package com.example.community.controller.advice;

import com.example.community.util.CommunityUtil;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.ControllerAdvice;
import org.springframework.web.bind.annotation.ExceptionHandler;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;

/**
 * @Author: huang
 * @Date: 2021/9/25 16:57
 * @Description:    @ControllerAdvice
 * 用于修饰类，表示该类是Controller的全局配置类。
 * 在此类中，可以对Controller进行如下三种全局配置： 异常处理方案、绑定数据方案、绑定参数方案。
 *
 * @ExceptionHandler
 * 用于修饰方法，该方法会在Controller出现异常后被调用，用于处理捕获到的异常。
 */
@ControllerAdvice(annotations = Controller.class)
public class ExceptionAdvice {

    private static final Logger logger = LoggerFactory.getLogger(Exception.class);

    /**
     *
     * @param e
     * @param request
     * @param response
     *          将异常写入日志，然后判断当前请求是需要返回页面还是json数据，从请求中获取请求头参数x-requested-with，
     *          如果参数等于XMLHttpRequest，那么请求是一个异步请求，返回一个json字符，否则重定向到error页面。
     */
    @ExceptionHandler({Exception.class})
    public void handlerException(Exception e, HttpServletRequest request, HttpServletResponse response) throws IOException {
        logger.error("服务器发送异常:" + e.getMessage());

        for (StackTraceElement element : e.getStackTrace()) {
            logger.error(element.toString());
        }

        String xRequestWith = request.getHeader("x-requested-with");
        if ("XMLHttpRequest".equals(xRequestWith)) {
            response.setContentType("application/plain;charset=utf-8");
            PrintWriter writer = response.getWriter();
            writer.write(CommunityUtil.getJSONString(1,"服务器异常!"));
        }else {
            response.sendRedirect(request.getContextPath() + "/error");
        }
    }
}
